# -*- coding:utf-8 -*-
# 北梦测教育
# 课程咨询加微信：xiaobeiceshi

# 数据预处理器

import ddddocr
from PIL import Image
import base64
from io import BytesIO

def dddd_ocr_text(data):
    """
    进行OCR识别的函数
    :return: 识别之后的文本
    """

    # 分割data文本，得到头部信息和 Base64编码部分
    encode_data = data.split(",")[1]


    # 解码 Base64 元数据
    decode_data = base64.b64decode(encode_data)

    # 把解码数据调整为图像数据
    image = Image.open(BytesIO(decode_data))

    # 把图像数据转成字节数据
    image_bytes = BytesIO()
    image.save(image_bytes,format="PNG")
    image_bytes = image_bytes.getvalue()

    # 初始化
    ocr = ddddocr.DdddOcr()
    text = ocr.classification(image_bytes)
    return text



if __name__ == '__main__':
    data = ""
    print(dddd_ocr_text(data))


